#! /bin/bash

source activate det && cd /root/code/vision/references/detection
torchrun --nproc_per_node=1 train.py\
    --model retinanet_resnet50_fpn --epochs 26 \
    --lr-steps 16 22 --aspect-ratio-group-factor 3 --lr 0.01 \
    --weights-backbone ResNet50_Weights.IMAGENET1K_V1 \
    --data-path="/root/code/mmdetection/data/coco" \
    --print-freq 1000 \
    --batch-size 16 --workers 8 --amp \
    --start_epoch 13 --resume /root/code/vision/references/detection/model_12.pth